#!/usr/bin/python
# -*- coding: UTF-8 -*-

from MiniBotControlLib import MiniBotControlLib

# ------------------------------
# 	1、连接设备 Connect
#   2、扫描模块ID ScanId
#   3、设置模块ID所属模块类别  SetMotorITGType
#   4、获取模块ID所属模块类别  GetMotorITGType
# 	5、断开设备 Disconnect
# ------------------------------

# 声明控制库对象
mlib = MiniBotControlLib("udp")


# *************************
#   1、连接设备
#*************************#
ip = "10.10.100.254"  
r = mlib.Connect(ip, 8899)
# 带超时的连接方式
#r = mlib.Connect(ip,8899,2000)
if(r > 0):
    print("连接成功")
else:
    print("连接失败")


# *************************
#   2、扫描模块ID
# *************************
motorId, motorLoc = mlib.ScanId(0, 50)
print("motorId", motorId)

# *************************
#   3、读取模块ID所属模块类别
# *************************
newId = motorId[0]
input("输入任意字符回车，获取设置模块" + str(newId) + "的类别")
itg = mlib.GetMotorITGType(newId)
if(itg == 2):
    print("模块" + str(newId) + "是I模块")
if(itg == 3):
    print("模块" + str(newId) + "是T模块")
if(itg == 4):
    print("模块" + str(newId) + "是G模块")

# *************************
#   4、设置模块ID所属模块类别
# *************************
input("输入任意字符回车，设置模块" + str(newId) + "为 G 模块")
r = mlib.SetMotorITGType(newId, 4)
if(r):
    print("设置成功")
else:
    print("设置失败")

# *************************
#   5、设置回原来参数
# *************************
print("设置回原来参数")
mlib.SetMotorITGType(newId, itg)

# *************************
# 	6、断开设备
# *************************
input("输入任意字符回车，断开连接")
r = mlib.Disconnect()
if(r > 0):
    print("断开成功")
else:
    print("断开失败")
